from profile_utils import (
    load_names_from_file,
    load_big5_from_file,
    generate_profile,
    generate_prompt
)

# 从文件加载人名
load_names_from_file("names.txt")

# 从文件加载预定义性格
load_big5_from_file("big5.json")

# 生成一个 profile
profile = generate_profile()

# 模拟一些交互记录
interactions = [
    {
        "partner_id": 1,
        "content": "I agree with your point to some extent.",
        "partner_belief": -0.5
    },
    {
        "partner_id": 2,
        "content": "I strongly disagree with your opinion.",
        "partner_belief": 1.2
    }
]

# 生成 prompt
prompt = generate_prompt(profile, interactions)

# 打印生成的 prompt
print(prompt)